import os
import matplotlib.pylab as plt


# 图像保存

# 参数解释
# Name:默认保存的图片名
# Keep:是否保存
# ImageTitle原图片的标题
# 默认保存为tif格式
def KeepImage(Name, Keep=True, ImageTitle='原图片标题未知'):
    if Keep == True:
        print('====================保存图片==================')
        # 判断是否存在图像
        # 如果已经存在
        if os.path.exists(Name + '.tif'):
            print('===========================')
            print('原图片的标题为:')
            print(ImageTitle)
            print('===========================')
            print('图像已经存在,是否继续保存?')
            reinput_1 = True
            response=' '
            while reinput_1:
                response = input('保存输入yes/y,不保存输入no/n:\n')
                R=response.upper()
                if R == 'Y'or R == 'YES'or R == 'NO'or R == 'N':
                    reinput_1 = False
            # 继续更名保存图片
            if (response.upper() == 'YES') or response.upper() == 'Y':
                reinput = True
                inputstr = ''
                while reinput:
                    inputstr = input('输入保存的名称,注意添加保存图片的类型，例如xxx.png:\n')
                    if inputstr != '':
                        reinput = False
                plt.savefig('./' + inputstr)
                print('图像' + inputstr + '保存成功！！')

        # 如果不存在,默认保存为tif格式
        else:
            plt.savefig('./' + Name + '.tif')
            print('图像' + Name + '.tif' + '保存成功！！')
